package com.afs.electronicforensics.utils;

import java.util.HashMap;
import java.util.Map;

public class FileTypeUtil {

    public final static Map<String, Integer> FILE_TYPE_MAP = new HashMap<String, Integer>();

    private FileTypeUtil() {}

    public static Integer getType(String url) {
        Integer integer = FILE_TYPE_MAP.get(url.toLowerCase());
        if(integer == null) {
            return 0;
        }
        return integer;
    }

    static
    {
        FILE_TYPE_MAP.put("bmp", 2);
        FILE_TYPE_MAP.put("jpg", 2);
        FILE_TYPE_MAP.put("jpeg", 2);
        FILE_TYPE_MAP.put("png", 2);
        FILE_TYPE_MAP.put("tiff", 2);
        FILE_TYPE_MAP.put("gif", 2);
        FILE_TYPE_MAP.put("pcx", 2);
        FILE_TYPE_MAP.put("tga", 2);
        FILE_TYPE_MAP.put("exif", 2);
        FILE_TYPE_MAP.put("fpx", 2);
        FILE_TYPE_MAP.put("svg", 2);
        FILE_TYPE_MAP.put("psd", 2);
        FILE_TYPE_MAP.put("cdr", 2);
        FILE_TYPE_MAP.put("pcd", 2);
        FILE_TYPE_MAP.put("dxf", 2);
        FILE_TYPE_MAP.put("ufo", 2);
        FILE_TYPE_MAP.put("eps", 2);
        FILE_TYPE_MAP.put("ai", 2);
        FILE_TYPE_MAP.put("raw", 2);
        FILE_TYPE_MAP.put("wmf", 2);
        /*FILE_TYPE_MAP.put("txt", 2);
        FILE_TYPE_MAP.put("doc", 2);
        FILE_TYPE_MAP.put("docx", 2);
        FILE_TYPE_MAP.put("xls", 2);
        FILE_TYPE_MAP.put("htm", 2);
        FILE_TYPE_MAP.put("html", 2);
        FILE_TYPE_MAP.put("jsp", 2);
        FILE_TYPE_MAP.put("rtf", 2);
        FILE_TYPE_MAP.put("wpd", 2);
        FILE_TYPE_MAP.put("pdf", 2);
        FILE_TYPE_MAP.put("ppt", 2);*/
        FILE_TYPE_MAP.put("mp4", 1);
        FILE_TYPE_MAP.put("avi", 1);
        FILE_TYPE_MAP.put("mov", 1);
        FILE_TYPE_MAP.put("wmv", 1);
        FILE_TYPE_MAP.put("asf", 1);
        FILE_TYPE_MAP.put("navi", 1);
        FILE_TYPE_MAP.put("3gp", 1);
        FILE_TYPE_MAP.put("mkv", 1);
        FILE_TYPE_MAP.put("f4v", 1);
        FILE_TYPE_MAP.put("rmvb", 1);
        FILE_TYPE_MAP.put("webm", 1);
        FILE_TYPE_MAP.put("mp3", 3);
        FILE_TYPE_MAP.put("wma", 3);
        FILE_TYPE_MAP.put("wav", 3);
        FILE_TYPE_MAP.put("mod", 3);
        FILE_TYPE_MAP.put("ra", 3);
        FILE_TYPE_MAP.put("cd", 3);
        FILE_TYPE_MAP.put("md", 3);
        FILE_TYPE_MAP.put("asf", 3);
        FILE_TYPE_MAP.put("aac", 3);
        FILE_TYPE_MAP.put("vqf", 3);
        FILE_TYPE_MAP.put("ape", 3);
        FILE_TYPE_MAP.put("mid", 3);
        FILE_TYPE_MAP.put("ogg", 3);
        FILE_TYPE_MAP.put("m4a", 3);
        FILE_TYPE_MAP.put("vqf", 3);
    }
}


